#pragma once

#include <pybind11/numpy.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <pybind11/stl_bind.h>

namespace py = pybind11;

namespace pytxdnn {

// txdnn : Data Type, kernel params, hardware abstraction
void bindTxdnn(py::module &txdnn);

// graph: Kernels in graph model
void bindTxdnnGraph(py::module &graph);

// eager: Kernels in eager mode
void bindTxdnnEager(py::module &eager);

} // namespace pytxdnn
